import tile
import os, pygame

class WorldManager():
    """Manages the game world"""
    
    def __init__(self, worldFile):
        self.read_world(worldFile)
        
    def draw(self, screen):
        #[[self.world[row][col].draw(screen) for col in range(len(self.world[0]))] for row in range(len(self.world))]
        
        self.permGroup.draw(screen)
        self.bgGroup.draw(screen)
        self.wallGroup.draw(screen)
        self.cannonGroup.draw(screen)
        
    def read_world(self, inputFile):
        #width and height of images (hard coded?) (read from 1 image and
        #applied expecting images to all be uniform?)
        pWidth = 32
        pHeight = 32
        
        #width and height (in tiles) of map from file
        width = 32
        height = 22
        
        #number key mapping from file
        self.mapKey = {
                0: pygame.image.load(os.path.join('imgs', 'grass.png')).convert(),
                1: pygame.image.load(os.path.join('imgs', 'wall-blue.png')).convert_alpha(),
                2: pygame.image.load(os.path.join('imgs', 'cannon-blue.png')).convert_alpha(),
                3: pygame.image.load(os.path.join('imgs', 'ground-blue.png')).convert()}
        
        #definitions of which keys belong to which group from file
        permGroupDef = []
        bgGroupDef = [0, 3]
        wallGroupDef = [1]
        cannonGroupDef = [2]
        
        #map representation from file
        mapInFile = [
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 0],
                     [0, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 0],
                     [0, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 0],
                     [0, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3, 3, 3, 3, 3, 3, 3, 3, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
        '''
        mapInFile = [
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [3, 3, 3, 3, 3, 3, 3, 3, 3, 3],
                     [0, 0, 0, 0, 0, 3, 3, 3, 3, 0],
                     [0, 3, 0, 3, 0, 3, 3, 3, 3, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                     [3, 3, 3, 3, 3, 3, 3, 3, 3, 3],
                     [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
        '''
        #set up groups for assignments during loop
        self.permGroup = pygame.sprite.Group()
        self.bgGroup = pygame.sprite.Group()
        self.wallGroup = pygame.sprite.Group()
        self.cannonGroup = pygame.sprite.Group()
        
        #init size of world array
        self.world = [[None for col in range(width)] for row in range(height)]
                
        #create tiles in world array
        for row in range(height):
            for col in range(width):
                self.world[row][col] = tile.Tile(self.mapKey[mapInFile[row][col]])
                
                #set tile's position using screen's width and height
                self.world[row][col].rect.topleft = (pWidth*col, pHeight*row)
                
                #put tile in group based on defined definitions
                if mapInFile[row][col] in permGroupDef:
                    self.permGroup.add(self.world[row][col])
                elif mapInFile[row][col] in bgGroupDef:
                    self.bgGroup.add(self.world[row][col])
                elif mapInFile[row][col] in wallGroupDef:
                    self.wallGroup.add(self.world[row][col])
                elif mapInFile[row][col] in cannonGroupDef:
                    self.cannonGroup.add(self.world[row][col])